from app import create_app
from app.models.mysql_db import db
from app.models.user import User
from app.utils.logger import get_logger

logger = get_logger(__name__)

def create_admin_user():
    """创建管理员用户"""
    app = create_app()
    
    with app.app_context():
        try:
            # 检查是否已存在管理员用户
            admin = User.query.filter_by(username='admin').first()
            if admin:
                logger.info("管理员用户已存在")
                return
            
            # 创建管理员用户
            admin = User(
                username='admin',
                email='admin@example.com',
                password='123456',
                role='admin',
                nickname='系统管理员'
            )
            
            db.session.add(admin)
            db.session.commit()
            
            logger.info("管理员用户创建成功")
            
        except Exception as e:
            logger.error(f"创建管理员用户失败: {str(e)}")
            db.session.rollback()

if __name__ == '__main__':
    create_admin_user() 